project_root="/home/gaolinfeng/projects/nlpcoding-task/task2"
find $project_root/bookcorpus -name 'cache-*' -type f -delete -print
source $project_root/venv/bin/activate
python $project_root/src/test_text_infilling.py \
          --data_path $project_root/bookcorpus \
          --tokenizer_name $project_root/tokenizer \
          --model_name_or_path facebook/bart-base \
          --mask_probability 0.15 \
          --max_seq_length 256 \
          --config_name facebook/bart-base \
          --checkpoint $project_root/checkpoint_2/epoch_0/pytorch_model.bin
